package datastructure.recursion;

/**
 * 8皇后问题,回溯算法
 * @author lbw
 */
public class Queue8 {
    /**
     * 定义max表示有多少皇后
     */
    int max = 8;
    /**
     * 定义数组array,保存皇后位置放置的结果，比如arr={0,4,7,5,2,6,1,3}
     */
    int[] array = new int[max];
    /**
     * 统计一共多少个解决方法
     */
    int count;
    public static void main(String[] args) {
        Queue8 queue8 = new Queue8();
        queue8.check(0);
        System.out.println(queue8.count);
    }

    /**
     * 放置第n个皇后
     */
    private void check(int n){
        if (n == max){
            print();
            count++;
            return;
        }
        //依次放入皇后，并判断是否冲突
        for (int i = 0; i < max ; i++){
            //先把当前皇后n，放到该行的第一列
            array[n] = i;
            //判断放置的位置是否冲突
            if (judge(n)){
                //不冲突，放n+1个，开始递归
                check(n+1);
            }
            //如果冲突，继续执行，将皇后放在本行的下一列
        }
    }

    /**
     * 查看放置的第n个皇后，是否和前面已经摆放的皇后冲突
     */
    private boolean judge(int n){
        for (int i = 0; i < n; i++) {
            //array[i] == array[n] 判断第n个皇后是否和前面的n-1个皇后在同一列
            //Math.abs(n - i) == Math.abs(array[n] - array[i]) 判断是否在同一斜线
            if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])){
                return false;
            }
        }
        return true;
    }
    /**
     * 写一个方法，可以将皇后摆放的位置输出
     */
    private void print(){
        for (int i = 0; i < array.length; i++) {
            System.out.print(array[i] + "");
        }
        System.out.println();
    }
}
